hacker rank math solution with python